import taichi as ti

# 初始化太极
ti.init(arch=ti.cpu)
# 42x63的张量，每个元素是数值
a = ti.field(dtype=ti.f32,shape=(42,63))
# 长度为4的张量，每个元素是一个向量
b = ti.Vector.field(n=3,dtype=ti.f32,shape=(4))
# 3x5的张量，每个元素是3x2的矩阵
c = ti.Matrix.field(n=3,m=2,dtype=ti.f32,shape=(3,5))

# 0维张量，表示一个数值
loss = ti.field(dtype=ti.f32,shape=())

a[3,4]=1
print("a[3,4]={}".format(a[3,4]))

b[2]=[6,7,8]
print("b[0]={} {} {}".format(b[2][0],b[2][1],b[2][2]))
print(b[2])

loss[None]=3
print(loss[None])